package com.xyhuo.leetcode;

import com.xyhuo.leetcode.common.ListNode;

import java.util.ArrayList;
import java.util.List;

public class RemoveElements {
    public ListNode removeElements(ListNode head, int val) {
        List<ListNode> list = new ArrayList<>();
        ListNode node = head;
        while (node != null) {
            if (node.val != val) {
                list.add(node);
            }
            node = node.next;
        }
        if (list.size() > 1) {
            for (int i = 0; i < list.size() - 1; i++) {
                list.get(i).next = list.get(i + 1);
            }
        }
        if (list.size() > 0) {
            list.get(list.size() - 1).next = null;
        }
        return list.isEmpty() ? null : list.get(0);
    }

}
